import java.util.ArrayList;
import java.util.List;

/**
 * 面试题54：二叉搜索树的第k大节点
 */
public class Offer_54 {
    int ans, k;

    /**
     * 方法二：倒序中序遍历
     * <p>
     * 时间复杂度：O(n)
     * <p>
     * 空间复杂度：O(n)
     */
    public int kthLargest(TreeNode root, int k) {
        this.k = k;
        inorder(root);
        return ans;
    }

    private void inorder(TreeNode root) {
        if (root == null || k == 0) {
            return;
        }
        inorder(root.right);
        if (--k == 0) {
            ans = root.val;
            return;
        }
        inorder(root.left);
    }

    /**
     * 方法一：中序遍历 + 辅助数组
     * <p>
     * 时间复杂度：O(n)
     * <p>
     * 空间复杂度：O(n)
     */
    public int kthLargest1(TreeNode root, int k) {
        List<Integer> list = new ArrayList<>();
        inorder(root, list);
        return list.get(list.size() - k);
    }

    private void inorder(TreeNode root, List<Integer> list) {
        if (root == null) {
            return;
        }
        inorder(root.left, list);
        list.add(root.val);
        inorder(root.right, list);
    }
}
